import sys
path="../"
sys.path.append(path)

from xyz_src.mploop import loop
import numpy as np

batch = 0
modeldir = "../../../Models/aladip_multitask"
r = np.arange(-25, 25, 0.5)

workdir = f"umb_{k0}"
k = np.ones(len(r))*k0
for idx in range(len(r)):
    loop("init_config.npz", idx, batch, modeldir, workdir, r, k, ncpus=1, maxK=-1, damp=50, temp=50)
